package com.zjj.algorithm.learning.leetcode.enum1;

import java.util.ArrayList;
import java.util.List;

/**
 * 204. 计数质数
 * <p>
 * 给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：n = 10
 * 输出：4
 * 解释：小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
 * 示例 2：
 * <p>
 * 输入：n = 0
 * 输出：0
 * 示例 3：
 * <p>
 * 输入：n = 1
 * 输出：0
 * <p>
 * <p>
 * 提示：
 * <p>
 * 0 <= n <= 5 * 106
 *
 * @author zjj_admin
 * @date 2023/1/9 10:09
 */
public class CountPrimes {


    public static void main(String[] args) {
        long start = System.currentTimeMillis();
        int res = countPrimes(5000000);
        long end = System.currentTimeMillis();

        System.out.println("res = " + res);
        System.out.println("用时：" + (end - start) + " ms");
    }


    /**
     * 5000000 万个 ，用时：115 ms
     *
     * @param n
     * @return
     */
    public static int countPrimes(int n) {

        if (n <= 2) {
            return 0;
        }
        if (n <= 3) {
            return 1;
        }
        if (n <= 5) {
            return 2;
        }
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = i;
        }
        arr[1] = 0;
        List<Integer> res = new ArrayList<>();
        res.add(2);
        res.add(3);
        res.add(5);
        int count = 3;
        int index;
        for (int i = 2; i < n; i++) {
            if (arr[i] == 0) {
                continue;
            }
            if (arr[i] > 5) {
                res.add(arr[i]);
                count++;
            }
            for (Integer re : res) {
                for (int j = 0; j < n; j++) {
                    index = re * j;
                    if (index > n - 1) {
                        break;
                    }
                    arr[index] = 0;
                }
            }
            res = new ArrayList<>();
        }
        return count;
    }


    public static int countPrimes2(int n) {
        int cnt = 0;
        for (int i = 2; i < n; i++) {
            if (isPrime(i)) {
                cnt++;
            }
        }
        return cnt;
    }

    private static boolean isPrime(int num) {
        int max = (int) Math.sqrt(num);
        for (int i = 2; i <= max; i++) {
            if (num % i == 0) {
                return false;
            }
        }
        return true;
    }

}
